import pandas as pd

import PD.PdUtil
import fileUtil
import 去除企宽


def radius掉线人次人数统计(batch):
    df = 去除企宽.get_df(batch, ['所属小区', 'account', 'offline_time'])
    # 创建一个新列，将 account 和 offline_time 组合成元组
    df['account_offline_time'] = df['account'] + df['offline_time']

    # 按所属小区分组并进行聚合操作
    result = df.groupby('所属小区').agg(
        掉线人次=('account_offline_time', lambda x: x.nunique()),
        掉线用户数=('account', lambda x: x.nunique())
    ).reset_index()
    PD.PdUtil.to_csv(去除企宽.getPath(batch), result)


if __name__ == '__main__':
    radius掉线人次人数统计('0301_0331')
